package com.itheima.service.impl;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.itheima.dao.SyslogDao;
import com.itheima.domain.Syslog;
import com.itheima.service.SyslogService;
import com.itheima.utils.MybatisUtils;
import org.apache.ibatis.session.SqlSession;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.List;

public class SyslogServiceImpl implements SyslogService {
    @Override
    public PageInfo findAll(Syslog syslog) {
        SqlSession sqlSession = MybatisUtils.getSqlSession();
        SyslogDao mapper = sqlSession.getMapper(SyslogDao.class);
        PageHelper.startPage(syslog.getPage(),syslog.getSize());
        List<Syslog> all = mapper.findAll(syslog);
        PageInfo pageInfo = new PageInfo(all);
        sqlSession.close();
        return pageInfo;
    }

    @Override
    public void delete(String[] ids) {
        SqlSession sqlSession = MybatisUtils.getSqlSession();
        SyslogDao mapper = sqlSession.getMapper(SyslogDao.class);
        mapper.delete(ids);
        sqlSession.close();
    }

    @Override
    public void save(Syslog syslog) {
        SqlSession sqlSession = MybatisUtils.getSqlSession();
        SyslogDao mapper = sqlSession.getMapper(SyslogDao.class);
        mapper.save(syslog);
        sqlSession.close();
    }

    @Override
    public ByteArrayOutputStream getReport() {
        List<Syslog> syslogList = null;
        //获取数据
        SqlSession sqlSession = MybatisUtils.getSqlSession();
        SyslogDao mapper = sqlSession.getMapper(SyslogDao.class);
        syslogList = mapper.findAll(new Syslog());
        sqlSession.close();

        //1.获取到对应的Excel文件，工作簿文件
        Workbook wb = new XSSFWorkbook();
        //2.创建工作表
        Sheet s = wb.createSheet("日志文件");
        //设置通用配置
        CellStyle cs_field = wb.createCellStyle();
        cs_field.setAlignment(HorizontalAlignment.CENTER);
        cs_field.setBorderTop(BorderStyle.THIN);
        cs_field.setBorderBottom(BorderStyle.THIN);
        cs_field.setBorderLeft(BorderStyle.THIN);
        cs_field.setBorderRight(BorderStyle.THIN);

        //制作标题
        s.addMergedRegion(new CellRangeAddress(1,1,1,12));
        Row row_1 = s.createRow(1);
        Cell cell_1_1 = row_1.createCell(1);
        cell_1_1.setCellValue("日志导出信息");
        //创建一个样式
        CellStyle cs_title = wb.createCellStyle();
        cs_title.setAlignment(HorizontalAlignment.CENTER);
        cs_title.setVerticalAlignment(VerticalAlignment.CENTER);
        cell_1_1.setCellStyle(cs_title);
        //制作表头
        String[] fields = {"ID","访问时间","访问用户","访问IP","资源URL",
                "执行时间","访问方法"};
        Row row_2 = s.createRow(2);

        for (int i = 0; i < fields.length; i++) {
            Cell cell_2_temp = row_2.createCell(1 + i); //++
            cell_2_temp.setCellValue(fields[i]);    //++
            cell_2_temp.setCellStyle(cs_field);
        }

        int row_index = 0;
        for (Syslog q : syslogList) {
            int cell_index = 0;
            Row row_temp = s.createRow(3 + row_index++);

            Cell cell_data_1 = row_temp.createCell(1 + cell_index++);
            cell_data_1.setCellValue(q.getId());    //++
            cell_data_1.setCellStyle(cs_field);

            Cell cell_data_2 = row_temp.createCell(1 + cell_index++);
            cell_data_2.setCellValue(q.getVisitTime());    //++
            cell_data_2.setCellStyle(cs_field);

            Cell cell_data_3 = row_temp.createCell(1 + cell_index++);
            cell_data_3.setCellValue(q.getUsername());    //++
            cell_data_3.setCellStyle(cs_field);

            Cell cell_data_4 = row_temp.createCell(1 + cell_index++);
            cell_data_4.setCellValue(q.getIp());    //++
            cell_data_4.setCellStyle(cs_field);

            Cell cell_data_5 = row_temp.createCell(1 + cell_index++);
            cell_data_5.setCellValue(q.getUrl());    //++
            cell_data_5.setCellStyle(cs_field);

            Cell cell_data_6 = row_temp.createCell(1 + cell_index++);
            cell_data_6.setCellValue(q.getExecutionTime());    //++
            cell_data_6.setCellStyle(cs_field);

            Cell cell_data_7 = row_temp.createCell(1 + cell_index++);
            cell_data_7.setCellValue(q.getMethod());    //++
            cell_data_7.setCellStyle(cs_field);
        }

        //将内存中的workbook数据写入到流中
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        try {
            wb.write(os);
            wb.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return os;
    }
}
